import chromadb

client = chromadb.Client()
customer_support_collection = client.create_collection(
    name="customer_support"
)

customer_support_collection.add(
    ids=["1", "2", "3"],
    documents=[
        "Toothbrushes can be returned up to 360 days after purchase if unopened.",
        "Shipping is free of charge for all orders.",
        "Shipping normally takes 2-3 business days"
    ]
)

user_query = "What is your return policy for tooth brushes?"

context = customer_support_collection.query(
    query_texts=[user_query],
    n_results=1
)['documents'][0]

print(context)
